import unittest
# 导入测试类
from test01 import TestAdd
from test02 import TestThree
# 创建一个测试套件对象
suite = unittest.TestSuite()
# 将测试用例添加到测试套件中
suite.addTest(TestAdd('test_add01'))
suite.addTest(TestAdd('test_add02'))
suite.addTest(TestAdd('test_add03'))
# 将Test02添加到测试套件中
suite.addTest(TestThree('test01'))
suite.addTest(TestThree('test02'))
suite.addTest(TestThree('test03'))


if __name__ == '__main__':
    # 执行测试套件
    # 创建测试运行程序
    runner = unittest.TextTestRunner()
    runner.run(suite)